Class DenseEncoder
Defined in File dense_encoder.py
Inheritance Relationships
Base Type
public encoder_decoder_model.cnn_basenet.CNNBaseModel
(Class CNNBaseModel)
Class Documentation
- encoder_decoder_model.dense_encoder.DenseEncoder : public encoder_decoder_model.cnn_basenet.CNNBaseModel
Public Functions
- __init__(self, l, n, growthrate, phase, with_bc=False, bc_theta=0.5)
Encoder based on DenseNet.
- Parameters
l – According to the DenseNet paper L refers to the depth of the network
growthrate – According to the DenseNet paper GrowthRate refers to the dense block out dims
n – According to the DenseNet N refers to the block nums of the network
with_bc – whether to use BC in the model
phase – is training or testing
bc_theta – transmition theta thresholding
- __str__(self)
- Returns
- encode(self, input_tensor, name)
DenseNet encoding.
- Parameters
input_tensor –
name –
- Returns